package com.ruoyi.attendance.mapper;

import com.ruoyi.attendance.dto.ScheduleQueryDTO;
import com.ruoyi.attendance.entity.ScheduleManagement;
import com.ruoyi.attendance.vo.ScheduleRecordVO;
import com.ruoyi.attendance.vo.ScheduleUserVO;
import org.apache.ibatis.annotations.MapKey;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;

import java.util.Date;
import java.util.List;
import java.util.Map;

@Mapper
public interface ScheduleManagementMapper {
    /**
     * 根据条件查询排班用户列表
     */
    // ScheduleManagementMapper.java
    Long countValidUsers(@Param("query") ScheduleQueryDTO query);
    // List<ScheduleUserVO> listScheduleUsers(@Param("query") ScheduleQueryDTO query);


    /**
     * 查询用户列表
     * @param query 查询参数
     * @return 用户列表
     */
    List<ScheduleUserVO> selectUserList(ScheduleQueryDTO query);

    /**
     * 查询用户排班记录
     * @param userId 用户ID
     * @param startDate 开始日期
     * @param endDate 结束日期
     * @return 排班记录列表
     */
    List<ScheduleRecordVO> selectScheduleListByUserId(
            @Param("userId") Integer userId,
            @Param("startDate") Date startDate,
            @Param("endDate") Date endDate
    );


    /**
     * 批量插入排班记录（修改为使用Entity类）
     */
    void batchInsertSchedules(@Param("records") List<ScheduleManagement> records);

    int updateScheduleManagement(@Param("record") ScheduleManagement scheduleManagement);

    ScheduleManagement getScheduleManagementByIdAndDate(@Param("id") Integer id,@Param("date") Date date);

    List<ScheduleManagement> getScheduleManagementByIdAndDateList(@Param("id") Integer id,@Param("date") Date date);
}